package com.heima.leetcode.practice;

import java.util.Arrays;

/**
 * @Classname E204
 * @Description leetcode 204. 计数质数
 * @Date 2025/10/13 14:06
 * @Author 勾新杰
 */
public class E204 {

    /**
     * 暴力解法
     *
     * @param n n
     * @return 质数个数
     */
    public int countPrimes1(int n) {
        int count = 0;
        for (int i = 1; i < n; i++) {
            if (isPrime(i)) count++;
        }
        return count;
    }

    /**
     * 是否是质数
     *
     * @param num 数
     * @return 是质数，返回true
     */
    private boolean isPrime(int num) {
        if (num < 2) return false;
        for (int i = 2; i * i <= num; i++) {
            if (num % i == 0) return false;
        }
        return true;
    }

    /**
     * 埃拉托斯特尼筛法 <br><br>
     * 思想：标记所有合数，剩余的就是质数
     *
     * @param n n
     * @return 质数个数
     */
    public int countPrimes2(int n) {
        if (n == 0) return 0;
        boolean[] isPrime = new boolean[n + 1];
        Arrays.fill(isPrime, true);
        isPrime[0] = false;
        isPrime[1] = false;
        for (int i = 2; i <= n; i++) {
            if (isPrime[i]) {
                for (int j = i + i; j <= n; j += i) {
                    isPrime[j] = false;
                }
            }
        }
        int count = 0;
        for (int i = 0; i < n; i++) {
            if (isPrime[i]) count++;
        }
        return count;
    }
}
